package correct.day5;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;

public class Test2 {
    static int[] dx = {0, 0, 1, -1};
    static int[] dy = {1, -1, 0, 0};
    public static int rotApple (ArrayList<ArrayList<Integer>> grid)
    {
        int m = grid.size();
        int n = grid.get(0).size();
        boolean[][] vis = new boolean[m][n];
        Queue<int[]> q = new LinkedList<>();
        for(int i = 0; i < m; i++)
        {
            for(int j = 0; j < n; j++)
            {
                if(grid.get(i).get(j) == 2)
                {
                    q.add(new int[]{i, j});
                }
            }
        }
        int ret = 0; // 计算感染次数
        while(!q.isEmpty())
        {
            int sz = q.size();
            // 判断一次感染，sz为此次感染有多少个腐烂苹果能传播感染, 1-1 2-2 3-2 4-1 5-0
            while(sz-- != 0)
            {
                int[] t = q.poll();
                int a = t[0], b = t[1];
                // 判断某个格子四个方向感染都感染一次
                for(int i = 0; i < 4; i++)
                {
                    int x = a + dx[i], y = b + dy[i];
                    if(x >= 0 && x < m && y >= 0 && y < n && vis[x][y] == false
                            && grid.get(x).get(y) == 1)
                    {
                        vis[x][y] = true;
                        q.add(new int[]{x, y});
                    }
                }
            }
            ret++;
        }
        // 判断剩余的苹果
        for(int i = 0; i < m; i++)
        {
            for(int j = 0; j < n; j++)
            {
                if(grid.get(i).get(j) == 1 && !vis[i][j])
                {
                    return -1;
                }
            }
        }
        return ret - 1;
    }

    // [[2,1,1],[1,0,1],[1,1,1]]
    // [2,1,1]
    // [1,0,1]
    // [1,1,1]
    public static void main(String[] args) {
        ArrayList<ArrayList<Integer>> arrayLists = new ArrayList<>();
        ArrayList<Integer> arrayList1 = new ArrayList<>();
        arrayList1.add(2);
        arrayList1.add(1);
        arrayList1.add(1);

        ArrayList<Integer> arrayList2 = new ArrayList<>();
        arrayList2.add(1);
        arrayList2.add(0);
        arrayList2.add(1);

        ArrayList<Integer> arrayList3 = new ArrayList<>();
        arrayList3.add(1);
        arrayList3.add(1);
        arrayList3.add(1);

        arrayLists.add(arrayList1);
        arrayLists.add(arrayList2);
        arrayLists.add(arrayList3);

        System.out.println(rotApple(arrayLists));
    }
}
